from agno.agent import Agent
# from agno.models.anthropic import Claude
from agno.models.openai import OpenAILike
from agno.tools.reasoning import ReasoningTools
from agno.tools.yfinance import YFinanceTools
from configs.basic_config import ALI_GENERATION_MODEL, AGENT_URL, ALI_DASHSCOPE_API_KEY


thinking_agent = Agent(
    # model=Claude(id="claude-3-7-sonnet-latest"),
    model=OpenAILike(id=ALI_GENERATION_MODEL, name=ALI_GENERATION_MODEL, api_key=ALI_DASHSCOPE_API_KEY,
                     base_url=AGENT_URL),
    tools=[
        ReasoningTools(add_instructions=True),
        YFinanceTools(
            stock_price=True,
            analyst_recommendations=True,
            company_info=True,
            company_news=True,
        ),
    ],
    instructions="Use tables where possible",
    markdown=True,
)

thinking_agent.print_response("写一篇报告对比下 NVDA 和 TSLA", stream=True)